/*

 * Licensed to the Apache Software Foundation (ASF) under one

 * or more contributor license agreements.  See the NOTICE file

 * distributed with this work for additional information

 * regarding copyright ownership.  The ASF licenses this file

 * to you under the Apache License, Version 2.0 (the

 * "License"); you may not use this file except in compliance

 * with the License.  You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

package com.bff.gaia.unified.runners.core.construction;



import com.google.auto.service.AutoService;

import com.bff.gaia.unified.model.pipeline.v1.RunnerApi;

import com.bff.gaia.unified.model.pipeline.v1.RunnerApi.FunctionSpec;

import com.bff.gaia.unified.sdk.runners.AppliedPTransform;

import com.bff.gaia.unified.sdk.transforms.Flatten;

import com.bff.gaia.unified.sdk.transforms.PTransform;

import com.bff.gaia.unified.sdk.transforms.windowing.Window.Assign;



import java.util.Collections;

import java.util.Map;



/**

 * Utility methods for translating a {@link Assign} to and from {@link RunnerApi} representations.

 */

public class FlattenTranslator implements PTransformTranslation.TransformPayloadTranslator<Flatten.PCollections<?>> {



  public static PTransformTranslation.TransformPayloadTranslator create() {

    return new FlattenTranslator();

  }



  private FlattenTranslator() {}



  @Override

  public String getUrn(Flatten.PCollections<?> transform) {

    return PTransformTranslation.FLATTEN_TRANSFORM_URN;

  }



  @Override

  public FunctionSpec translate(

      AppliedPTransform<?, ?, Flatten.PCollections<?>> transform, SdkComponents components) {

    return RunnerApi.FunctionSpec.newBuilder().setUrn(getUrn(transform.getTransform())).build();

  }



  /** Registers {@link FlattenTranslator}. */

  @AutoService(TransformPayloadTranslatorRegistrar.class)

  public static class Registrar implements TransformPayloadTranslatorRegistrar {

    @Override

    public Map<? extends Class<? extends PTransform>, ? extends PTransformTranslation.TransformPayloadTranslator>

        getTransformPayloadTranslators() {

      return Collections.singletonMap(Flatten.PCollections.class, new FlattenTranslator());

    }

  }

}